"""
项目配置文件
"""
import os
from pathlib import Path

# 项目根目录
ROOT_DIR = Path(__file__).parent.parent

# 文档目录
DOCUMENTS_DIR = os.path.join(ROOT_DIR, "documents")

# 数据库目录
CHROMA_DB_DIR = os.path.join(ROOT_DIR, "chroma_db")

# 向量存储配置
COLLECTION_NAME = "rag_collection"

# 创建必要的目录
os.makedirs(DOCUMENTS_DIR, exist_ok=True)
os.makedirs(CHROMA_DB_DIR, exist_ok=True) 